# -*- coding: utf-8 -*-

# Copyright (c) 2010 Jérémie DECOCK (http://www.jdhp.org)

import random

class NaiveAgent:

    state_list = None
    action_list = None

    state_history = []
    action_history = []
    reward_history = []

    def __init__(self):
        pass

    def get_action(self, state, reward):
        action = random.choice(self.action_list)

        self.state_history.append(state)
        self.reward_history.append(reward)
        self.action_history.append(action)

        return action

